package com.github.yangyishe.p400;

import java.util.Stack;

/**
 * 394. 字符串解码
 * https://leetcode.cn/problems/decode-string/description/?envType=study-plan-v2&envId=top-100-liked
 *
 * 给定一个经过编码的字符串，返回它解码后的字符串。
 *
 * 编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
 *
 * 你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。
 *
 * 此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。
 *
 *
 *
 * 示例 1：
 *
 * 输入：s = "3[a]2[bc]"
 * 输出："aaabcbc"
 * 示例 2：
 *
 * 输入：s = "3[a2[c]]"
 * 输出："accaccacc"
 * 示例 3：
 *
 * 输入：s = "2[abc]3[cd]ef"
 * 输出："abcabccdcdcdef"
 * 示例 4：
 *
 * 输入：s = "abc3[cd]xyz"
 * 输出："abccdcdcdxyz"
 *
 *
 * 提示：
 *
 * 1 <= s.length <= 30
 * s 由小写英文字母、数字和方括号 '[]' 组成
 * s 保证是一个 有效 的输入。
 * s 中所有整数的取值范围为 [1, 300]
 */
public class Problem394 {
    public static void main(String[] args) {
        String s="abc3[cd]xyz";

        Problem394 problem394 = new Problem394();
        String decodeString = problem394.decodeString(s);
        System.out.println("decodeString = " + decodeString);

    }

    /**
     * 思路：
     * 此题的解决，与逆波兰表达式十分相似。故优先考虑使用栈解决。
     *
     * 使用两个栈，分别记录数字与[]内部的内容。
     * 收集内容时，如果遇到字母，则表示为字符部，否则为数组部。
     *
     *
     *
     * @param s
     * @return
     */
    public String decodeString(String s) {


        StringBuilder before=new StringBuilder();
        int slow=0;
        char beforeType=' ';
        for (int fast=0;fast<s.length();fast++) {
            if(isLetter(s.charAt(fast))){
                // 处理字母
                if(beforeType=='l'){
                    continue;
                }else{
                    dealContent(s,beforeType,slow,fast-1);
                    slow=fast;
                }

                beforeType='l';
            }else if(isNumber(s.charAt(fast))){
                // 处理数字
                if(beforeType=='n'){
                    continue;
                }else{
                    dealContent(s,beforeType,slow,fast-1);
                    slow=fast;
                }

                beforeType='n';
            }else if(s.charAt(fast)=='['){
                // 处理前括号
                dealContent(s,beforeType,slow,fast-1);
                slow=fast;
                beforeType='[';
            }else{
                // 处理后括号
                dealContent(s,beforeType,slow,fast-1);
                slow=fast;
                beforeType=']';
            }

        }
        dealContent(s,beforeType,slow,s.length()-1);


        return currentSb.toString();
    }

    private boolean isLetter(char c){
        return c>='a'&&c<='z';
    }

    private boolean isNumber(char c){
        return c>='0'&&c<='9';
    }

    private Stack<String> strStack=new Stack<>();
    private StringBuilder currentSb =new StringBuilder();

    private Stack<Integer> numStack=new Stack<>();

//    private int currentTimes=1;

    /**
     * 处理内容，本质是处理各层级的当前字符串
     * 重点是梳理好字符串的层级结构
     * @param s
     * @param type
     * @param start
     * @param end
     */
    private void dealContent(String s,char type,int start,int end){
        //处理内容
        switch (type){
            // 通常优先级最高
            case 'n':
                String numStr = s.substring(start, end + 1);
                numStack.push(Integer.parseInt(numStr));
                break;
            case '[':
                strStack.push(currentSb.toString());
                currentSb=new StringBuilder();
                break;
            case 'l':
                String letStr=s.substring(start,end+1);
                currentSb.append(letStr);
                break;
            case ']':
                // 先处理好当前字符串的情况
                int currentTimes=numStack.isEmpty()?1:numStack.peek();

                StringBuilder pop = new StringBuilder(strStack.pop());
                for(int i=0;i<currentTimes;i++){
                    pop.append(currentSb.toString());
                }
                currentSb=pop;
                numStack.pop();
                break;


        }
    }
}
